// test_common.h
#ifndef _TEST_COMMON_H
#define _TEST_COMMON_H

#include <kunit/test.h>
#include "../hisi_l0_mem_pool.h"

struct test_context {
    int test_node_id;
    atomic_t alloc_count;
    atomic_t free_count;
};

// Function to get the global test context
struct test_context *get_test_context(void);

// Add these function declarations
int l0_mem_pool_test_init(void *test);
void l0_mem_pool_test_exit(void *test);

#endif /* _TEST_COMMON_H */